package com.example.utils;

import java.util.LinkedHashSet;
import java.util.Random;
import java.util.Set;

public class RandomUtils {

    /***
     * 在0到max-1个数字中，随机抽取一个数字.
     * @param max
     * @return
     */
    public static int random(int max) {
        Random random = new Random();
        return random.nextInt(max);
    }

    /***
     *
     * 在0到max个数字中，抽取N个不重复的数字.
     * @param max
     * @param N
     * @return
     */
    public static Set<Integer> random(int max, int N) {
        Set<Integer> data = new LinkedHashSet<Integer>();
        Random random = new Random();
        while (data.size() != N) {
            int randInt = random.nextInt(max);
            if (!data.contains(randInt)) {
                data.add(randInt);
            }
        }
        return data;
    }
}
